package com.weizhizs.assistant.code.template;

import com.weizhizs.assistant.code.template.data.ApiTestData;
import com.weizhizs.assistant.code.template.data.TestControllerData;
import com.weizhizs.assistant.code.template.data.TestServiceData;

import java.util.HashMap;
import java.util.Map;

public class TestDataRegister {

    private static Map<String, TestDataLoador> generatorMap = new HashMap<>();

    static {
        generatorMap.put("TestService文件生成器", new TestServiceData());
//        generatorMap.put("TestController文件生成器", new TestControllerData());
        generatorMap.put("ApiTest文件生成器", new ApiTestData());
    }

    public static Map<String, TestDataLoador> templateDataMap() {
        return generatorMap;
    }
}
